package com.hencoderplus.demo1;

import android.media.AudioFormat;
import android.media.AudioRecord;
import android.media.MediaRecorder;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;

/**
 * hencoderplus
 * Created by lzw on 2018/10/30. 14:56:32
 * 邮箱：632393724@qq.com
 * All Rights Saved! Chongqing AnYun Tech co. LTD
 */
public class MyAudioRecord implements Runnable {

    private int mRecordBufferSize = 0;
    private AudioRecord mRecordInstance = null;
    private ByteArrayOutputStream saveVoiceByte;
    private boolean isRecord = false;
    private static int sampleRateInHz = 16000;//采样率
    // AudioName裸音频数据文件
    private static final String AudioName = "/sdcard/raw.raw";
    // NewAudioName可播放的音频文件
    private static final String NewAudioName = "/sdcard/new.wav";

    public MyAudioRecord(boolean b) {
        this.isRecord = b;
    }

    public void stopAudioRecord(){
        if (mRecordInstance != null) {
            mRecordInstance.stop();
            mRecordInstance.release();
            mRecordInstance = null;
            isRecord = false;
        }

    }
    @Override
    public void run() {
        StartRecord();// 开始录音
        copyWaveFile(AudioName, NewAudioName);// 得到音频文件

    }

    private void StartRecord() {
        int channelConfiguration = AudioFormat.CHANNEL_IN_MONO;
        int audioEncodingBits = AudioFormat.ENCODING_PCM_16BIT;
        mRecordBufferSize = AudioRecord.getMinBufferSize(16000, channelConfiguration, audioEncodingBits);
        mRecordInstance = new AudioRecord(MediaRecorder.AudioSource.MIC, 16000, channelConfiguration, audioEncodingBits,
                mRecordBufferSize);
        mRecordInstance.getChannelCount();
        FileOutputStream fos = null;
        File file = new File(AudioName);
        if (file.exists()) {
            file.delete();
        }
        try {
            fos = new FileOutputStream(file);
        } catch (FileNotFoundException e1) {
            // TODO Auto-generated catch block
            e1.printStackTrace();
        }

        if (mRecordInstance != null && mRecordInstance.getState() == AudioRecord.STATE_INITIALIZED) {
            try {
                mRecordInstance.startRecording();
            } catch (IllegalStateException e) {
                e.printStackTrace();
            }
        }
        byte[] audiodata = new byte[mRecordBufferSize];
        int readsize = 0;
        try {
            saveVoiceByte = new ByteArrayOutputStream();
        } catch (Exception e) {
            e.printStackTrace();
        }
        while (isRecord == true) {
            readsize = mRecordInstance.read(audiodata, 0, mRecordBufferSize);// ??
            if (AudioRecord.ERROR_INVALID_OPERATION != readsize) {
                try {
                    saveVoiceByte.write(audiodata, 0, readsize);// saveVoiceByte得到的字节流
                    fos.write(audiodata, 0, readsize);// 写入文件
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
        }
        try {
            saveVoiceByte.close();//
            fos.close();
        } catch (IOException e) {
            e.printStackTrace();
        }


    }


    private void copyWaveFile(String inFilename, String outFilename) {
        FileInputStream in = null;
        FileOutputStream out = null;
        long totalAudioLen = 0;
        long totalDataLen = totalAudioLen + 36;
        long longSampleRate = sampleRateInHz;
        int channels = 1;
        long byteRate = 16 * sampleRateInHz * channels / 8;
        byte[] data = new byte[mRecordBufferSize];
        try {
            in = new FileInputStream(inFilename);
            out = new FileOutputStream(outFilename);
            totalAudioLen = in.getChannel().size();
            totalDataLen = totalAudioLen + 36;
            WriteWaveFileHeader(out, totalAudioLen, totalDataLen, longSampleRate, channels, byteRate);
            while (in.read(data) != -1) {
                out.write(data);
            }
            in.close();
            out.close();
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 添加头信息，否则无法播放。
     */
    private void WriteWaveFileHeader(FileOutputStream out, long totalAudioLen, long totalDataLen, long longSampleRate,
                                     int channels, long byteRate) throws IOException {
        byte[] header = new byte[44];
        header[0] = 'R';
        header[1] = 'I';
        header[2] = 'F';
        header[3] = 'F';
        header[4] = (byte) (totalDataLen & 0xff);
        header[5] = (byte) ((totalDataLen >> 8) & 0xff);
        header[6] = (byte) ((totalDataLen >> 16) & 0xff);
        header[7] = (byte) ((totalDataLen >> 24) & 0xff);
        header[8] = 'W';
        header[9] = 'A';
        header[10] = 'V';
        header[11] = 'E';
        header[12] = 'f';
        header[13] = 'm';
        header[14] = 't';
        header[15] = ' ';
        header[16] = 16;
        header[17] = 0;
        header[18] = 0;
        header[19] = 0;
        header[20] = 1;
        header[21] = 0;
        header[22] = (byte) channels;
        header[23] = 0;
        header[24] = (byte) (longSampleRate & 0xff);
        header[25] = (byte) ((longSampleRate >> 8) & 0xff);
        header[26] = (byte) ((longSampleRate >> 16) & 0xff);
        header[27] = (byte) ((longSampleRate >> 24) & 0xff);
        header[28] = (byte) (byteRate & 0xff);
        header[29] = (byte) ((byteRate >> 8) & 0xff);
        header[30] = (byte) ((byteRate >> 16) & 0xff);
        header[31] = (byte) ((byteRate >> 24) & 0xff);
        header[32] = (byte) (2 * 16 / 8); // block align
        header[33] = 0;
        header[34] = 16; // bits per sample
        header[35] = 0;
        header[36] = 'd';
        header[37] = 'a';
        header[38] = 't';
        header[39] = 'a';
        header[40] = (byte) (totalAudioLen & 0xff);
        header[41] = (byte) ((totalAudioLen >> 8) & 0xff);
        header[42] = (byte) ((totalAudioLen >> 16) & 0xff);
        header[43] = (byte) ((totalAudioLen >> 24) & 0xff);
        out.write(header, 0, 44);
    }


}
